import secrets
import os


def generate_secret_key(length=32):
    """生成一个安全的密钥"""
    return secrets.token_urlsafe(length)


def save_key_to_env(file_name, key, value):
    """将密钥保存到 .env 文件"""
    with open(file_name, 'a' if os.path.exists(file_name) else 'w') as file:
        file.write(f'{key}={value}\n')


# 主执行逻辑
if __name__ == "__main__":
    secret_key = generate_secret_key()
    save_key_to_env('.env', 'JWT_SECRET_KEY', secret_key)
    print("SECRET_KEY 已生成并保存到 .env 文件中")
